
import time
name = time.strftime('%m-%d %H:%m:%S', time.localtime())

TRAIN_DIR = 'runs/no_sigmoid/' + name
VAL_DIR = 'runs/no_sigmoid/' + name

ACT_FUNC_IN_POSE = None


base_path = '/home/dl/datasets/DeepFashion/Category and Attribute Prediction Benchmark/'


NUM_EPOCH = 20
LEARNING_RATE = 0.0001
BATCH_SIZE = 16

WEIGHT_ATTR_NEG = 0.1
WEIGHT_ATTR_POS = 1
WEIGHT_LANDMARK_VIS_NEG = 0.5
WEIGHT_LANDMARK_VIS_POS = 0.5


# LOSS WEIGHT
WEIGHT_LOSS_CATEGORY = 1
WEIGHT_LOSS_ATTR = 20
WEIGHT_LOSS_LM_VIS = 4
WEIGHT_LOSS_LM_POS = 100


# VAL
VAL_CATEGORY_TOP_N = (1, 3, 5)
VAL_LM_RELATIVE_DIS = 0.1
